package com.atguigu.java.ai.langchain4j.tool;

import dev.langchain4j.agent.tool.P;
import dev.langchain4j.agent.tool.Tool;
import dev.langchain4j.agent.tool.ToolMemoryId;
import org.springframework.stereotype.Component;

@Component
public class CalculatorTools {
    @Tool/*(name = "加法运算", value = "将两个数字相加")*/
    double sum(
            @ToolMemoryId int memoryId,
            @P(value = "a", required = true) double a,
            @P(value = "b", required = true) double b
    ) {
        System.out.println("调用加法运算 " + memoryId);
        return a + b;
    }

    @Tool/*(name = "平方根运算", value = "计算一个数的平方根")*/
    double squareRoot(@ToolMemoryId int memoryId,double x) {
        System.out.println("调用加法运算 " + memoryId);
        return Math.sqrt(x);
    }
}